#!/bin/sh /etc/rc.common

START=15
. /lib/functions/network.sh
. /lib/functions.sh

proto="4"
[ -f /proc/net/ipv6_route ] && proto="4 6"

config_load freifunk-policyrouting
config_get enable pr enable
config_get fallback pr fallback
config_get zones pr zones

olsrd_rmtables() {
	# Remove custom routing tables from olsrd
	if [ "`uci -q get olsrd.@olsrd[0].RtTable`" == "111" ] || [ "`uci -q get olsrd.@olsrd[0].RtTableDefault`" == "112" ]; then
		uci delete olsrd.@olsrd[0].RtTable
		uci delete olsrd.@olsrd[0].RtTableDefault
		uci commit
	fi
}

olsrd_intalltables() {
	if [ ! "`uci -q get olsrd.@olsrd[0].RtTable`" == "111" ] || [ ! "`uci -q get olsrd.@olsrd[0].RtTableDefault`" == "112" ]; then
		uci set olsrd.@olsrd[0].RtTable='111'
		uci set olsrd.@olsrd[0].RtTableDefault='112'
		uci commit olsrd
		/etc/init.d/olsrd restart 2&> /dev/null
	fi
}

rt_tables() {
	tables="/etc/iproute2/rt_tables"
	if [ -z "`grep "110" $tables`" ]; then
		echo "110 localnets" >> $tables
	fi
	if [ -z "`grep "111" $tables`" ]; then
		echo "111 olsr" >> $tables
	fi
	if [ -z "`grep "112" $tables`" ]; then
		echo "112 olsr-default" >> $tables
	fi

}

disable_dyngw() {
	dyngwlib=`uci show olsrd |grep dyn_gw.so |awk {' FS="."; print $1"."$2 '}`
	if [ -n "$dyngwlib" ]; then
		uci set $dyngwlib.ignore=1
		uci commit olsrd
	fi

	dyngwplainlib=`uci show olsrd |grep dyn_gw_plain |awk {' FS="."; print $1"."$2 '}`
	if [ -n "$dyngwplainlib" ]; then
		rttable="$(uci -q get $dyngwplainlib.RtTable)"
		if [ -z "$rttable" ] || [ "$rttable" = "254" ]; then 
			uci set $dyngwplainlib.ignore=1
			uci commit olsrd
		fi
	fi
}

restart_services() {
	logger -s -t policyrouting -p info "Restarting services"
	/etc/init.d/network restart 2&> /dev/null
	/etc/init.d/olsrd restart 2&> /dev/null
}

boot() {
	if [ "$enable" = "1" ]; then
		[ -d /var/state ] || mkdir -p /var/state
		touch /var/state/freifunk-policyrouting
		start noservicerestart
	else
		olsrd_rmtables
	fi
}

add_lookup_rule() {
	name=${1/-/_}
	lookup=$2
	prio=$3

	if [ -z "$name" -o -z "$lookup" -o -z "$prio" ]; then
		logger -s -t policyrouting "Missing parameters for add_rule!"
	else
		for p in $proto; do
			if [ "$p" = "6" ]; then
				rule="rule6"
			else
				rule="rule"
			fi

			uci batch <<- EOF
				set network.${name}ipv${p}="$rule"
				set network.${name}ipv${p}.lookup="$lookup"
				set network.${name}ipv${p}.priority="$prio"
			EOF
		done
	fi
}

del_lookup_rule() {
	name=${1/-/_}
	for p in $proto; do
		uci -q delete network.${name}ipv${p}
	done
}

start() {
	if [ $enable = "1" ]; then
		logger -s -t policyrouting "Starting policy routing."
		rt_tables
		olsrd_intalltables
		disable_dyngw

		add_lookup_rule olsr olsr 1000
		add_lookup_rule localnets localnets 2000

		if [ "$fallback" = 1 ]; then
			add_lookup_rule olsr-default olsr-default 100000
		fi
	fi
	uci commit network
	if [ ! "$1" = "noservicerestart" ]; then
		restart_services
	fi
}

stop() {
	logger -s -t policyrouting "Stopping policy routing"
	olsrd_rmtables
	del_lookup_rule olsr-default
	del_lookup_rule olsr
	del_lookup_rule localnets
	uci commit network
	restart_services
	echo "Hint: To completely disable freifunk-policyrouting set enable=0 in /etc/config/freifunk-policyrouting."
}

restart() {
	logger -s -t policyrouting "Restarting policy routing"
	olsrd_rmtables
	del_lookup_rule olsr-default
	del_lookup_rule olsr
	del_lookup_rule localnets
	uci commit network
	start
}
